export PYTHONPATH=./zipformer:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="7"

input_test_dir="aishell aishell2 asru_test speechio_0 speechio_1 speechio_2 speechio_3 aishell2 speechio_4 speechio_5 speechio_6 speechio_7 speechio_8 speechio_9 speechio_10 test_meeting test_net speechio_11 speechio_12 speechio_13 speechio_14 speechio_15 speechio_16 speechio_17 speechio_18 speechio_19 speechio_20 speechio_21 speechio_22 speechio_23 speechio_24 speechio_25 speechio_26"
# input_test_dir="speechio_11 speechio_12 speechio_13 speechio_14 speechio_15 speechio_16 speechio_17 speechio_18 speechio_19 speechio_20 speechio_21 speechio_22 speechio_23 speechio_24 speechio_25 speechio_26"

# input_test_dir="speechio_7 speechio_8 speechio_9 speechio_10 test_net"
input_test_dir='aishell'
# 已完成：
# u2
./zipformer_streaming_u2_xlgeng/ctc_decode_hw.py \
    --epoch 10 \
    --avg 4 \
    --exp-dir ./zipformer_streaming_u2_xlgeng/exp-u2-stage04 \
    --train-switch u2 \
    --use-ctc 1 \
    --use-attention-decoder 1 \
    --max-duration 20 \
    --decoding-method attention-decoder-rescoring \
    --beam-size 32 \
    --ctc-weight 0.3 \
    --bpe-model data/lang_char_bpe/bpe.model \
    --lang-dir data/lang_char_bpe \
    --input-test-dir "$input_test_dir" \
    --test-parent-dir /home/local_data/yhdai/workspace/icefall/egs/multi_zh_en/ASR/data/dev_test_cuts/test_family_41 \
    --use-averaged-model false \
    --causal 1 \
    --chunk-size -1 \
    --left-context-frames -1

# --num-encoder-layers 2,2,3,4,3,2 \
# --feedforward-dim 512,768,1024,1536,1024,768 \
# --encoder-dim 192,256,384,512,384,256 \
# --encoder-unmasked-dim 192,192,256,256,256,192 \
# --valid-fbank-dir /home/work_nfs13/yhdai/workspace/icefall/egs/multi_zh_en/ASR/data/dev_test_cuts/dev_family \
# --valid_prefix valid_01 \
# 所有的测试集目录：
#/home/work_nfs13/yhdai/workspace/icefall/egs/multi_zh_en/ASR/data/dev_test_cuts/test_family \
# ctc-greedy-search
# attention-decoder-rescoring-no-ngram

# 默认值：
# --feedforward-dim 2,2,3,4,3,2
# --feedforward-dim 512,768,1024,1536,1024,768 \
# --encoder-dim 192,256,384,512,384,256 \
# --encoder-unmasked-dim 192,192,256,256,256,192 \
